package com.studyyang.confirm;

import com.rabbitmq.client.*;
import com.studyyang.utils.RabbitmqUtils;
import com.studyyang.utils.constant.RabbitMqConstant;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.TimeoutException;

/**
 * 气象局:测试消息确认机制
 * <p>
 * 模拟天气消息生产者
 *
 * @Author yangjin
 * @Description
 * @Date 2021/10/29/23:02
 **/
public class MetOffice {

    public static void main(String[] args) throws IOException, TimeoutException {
        //mock数据
        Map<String, String> mockData = new HashMap<>();
        mockData.put("china.henan.nanyang.20211212", "中国河南南阳-小雨-3至10摄氏度");
        mockData.put("china.hebei.xingtai.20211212", "中国河北邢台-多云-5至12摄氏度");
        mockData.put("china.shandong.jinan.20211212", "中国山东济南-晴-7至14摄氏度");
        mockData.put("us.jz.ls.20211212", "美国加州洛杉矶-阴-7至14摄氏度");
        mockData.put("us.xx.ny.20211212", "美国纽约-阴-7至14摄氏度");

        mockData.put("china.henan.nanyang.20211213", "中国河南南阳-小雨-3至10摄氏度");
        mockData.put("china.hebei.xingtai.20211213", "中国河北邢台-多云-5至12摄氏度");
        mockData.put("china.shandong.jinan.20211213", "中国山东济南-晴-7至14摄氏度");
        mockData.put("us.jz.ls.20211213", "美国加州洛杉矶-阴-7至14摄氏度");
        //获取mq连接
        Connection mqConn = RabbitmqUtils.getRabbitMqConn();
        //获取信道连接
        Channel channel = mqConn.createChannel();
        //开启消息监听
        channel.addConfirmListener(new ConfirmListener() {
            //处理以及被mq接收的消息

            /**
             * handleAck(long l, boolean b)
             * @param l 消息的唯一标识
             * @param b 是否进行批量接收消息
             * @throws IOException
             */
            @Override
            public void handleAck(long l, boolean b) throws IOException {
                System.out.println("消息已经被Broker接收，Tag：" + l);
            }

            //处理以及未被mq接收的消息
            /**
             * handleNack(long l, boolean b)
             * @param l 消息的唯一标识
             * @param b 是否进行批量接收消息
             * @throws IOException
             */
            @Override
            public void handleNack(long l, boolean b) throws IOException {
                System.out.println("消息已经被Broker拒收，Tag：" + l);
            }
        });
        //增加消息监听器
        /**
         * 这两个方法作用一样，均可监听获取返回的消息
         * addReturnListener(ReturnListener r) 不方便
         * addReturnListener(ReturnCallBack r) 方便
         */
        channel.addReturnListener(returnCallback -> {
            System.out.println("===============开始输出监听得到的返回的消息信息==============");
            System.out.println("错误编码：" + returnCallback.getReplyCode());
            System.out.println("错误描述：" + returnCallback.getReplyText());
            System.out.println("该消息绑定的交换机：" + returnCallback.getExchange());
            System.out.println("该消息绑定的路由键：" + returnCallback.getRoutingKey());
            System.out.println("消息主题：" + new String(returnCallback.getBody()));
            System.out.println("===============监听得到的返回的消息信息输出完毕==============");
        });
        //注意：这里就不用在声明队列了，消息生产者直接和交换机建立信道，不绑定队列
        //发送消息
        for (Map.Entry<String, String> entry : mockData.entrySet()) {
            //发布消息时，绑定交换机，不用绑定队列
            /**
             * basicPublish
             * 1、交换机名称
             * 2、routingKey，相当于消息筛选队列的条件
             * 3、mandtory，true-代表如果消息未被mq接收，则return返回回去，false-代表消息未被mq接收，则直接删除消息
             * 4、一些其他的属性
             * 5、消息的字节数组
             */
            channel.basicPublish(RabbitMqConstant.TOPIC_EXCHANGE_WEATHER, entry.getKey(),true, null, entry.getValue().getBytes(StandardCharsets.UTF_8));
        }
        System.out.println("天气消息发送成功!");

        //需要对队列进行监听，不能够关闭队列
       /* channel.close();
        mqConn.close();*/
    }
}
